How can I mock out this python function properly